"""
函数综合案例
黑马ATM
"""
money = 5000000  # 全局变量
name = "admin"
password = "admin123"
def login():
    """登录函数"""
    global name, password  # 声明name和password是全局变量
    input_name = input("请输入用户名: ")
    input_password = input("请输入密码: ")
    if input_name == name and input_password == password:
        print("登录成功!")
        return True  # 登录成功返回True
    else:
        print("用户名或密码错误!")
        return False  # 登录失败返回False
def check_balance():
    """查询余额函数"""
    global money  # 声明money是全局变量
    print(f"当前余额为: {money}元")
def deposit():
    """存款函数"""
    global money  # 声明money是全局变量
    amount = float(input("请输入存款金额: "))
    if amount > 0:
        money += amount
        print(f"存款成功! 当前余额为: {money}元")
    else:
        print("存款金额必须大于0!")
def withdraw():
    """取款函数"""
    global money  # 声明money是全局变量
    amount = float(input("请输入取款金额: "))
    if amount > 0:
        if amount <= money:
            money -= amount
            print(f"取款成功! 当前余额为: {money}元")
        else:
            print("余额不足!")
    else:
        print("取款金额必须大于0!")
def main():
    """主函数"""
    if login():  # 如果登录成功
        while True:
            print("""
            ====== 黑马ATM ======
            1. 查询余额
            2. 存款
            3. 取款
            4. 退出
            =====================
            """)
            choice = input("请输入操作选项(1-4): ")
            if choice == '1':
                check_balance()
            elif choice == '2':
                deposit()
            elif choice == '3':
                withdraw()
            elif choice == '4':
                print("感谢使用黑马ATM, 再见!")
                break  # 退出循环, 结束程序
            else:
                print("无效的操作选项, 请重新输入!")
    else:
        print("登录失败, 无法使用ATM服务!")
if __name__ == "__main__":
    main()  # 调用主函数